package com.lxy.snake;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.KeyEvent;
import java.awt.event.KeyListener;
import java.util.Random;

public class GameJpanel extends JPanel implements KeyListener, ActionListener {
    int length;
    int[] snakeX = new int[25 * 34];
    int[] snakeY = new int[25 * 24];
    int foodX;
    int foodY;
    Random random = new Random();
    String fx;
    int score;
    boolean fail;
    boolean start;
    Timer timer = new Timer(100, this);

    public GameJpanel() {
        init();
        setFocusable(true);
        this.addKeyListener(this);
        timer.start();
    }

    public void init() {
        length = 3;
        snakeX[0] = 100;
        snakeY[0] = 100;
        snakeX[1] = 75;
        snakeY[1] = 100;
        snakeX[2] = 50;
        snakeY[2] = 100;
        fx = "R";
        foodX = 25 + 25 * random.nextInt(34);
        foodY = 75 + 25 * random.nextInt(24);
        score = 0;
        fail = false;
        start = false;
    }

    @Override
    protected void paintComponent(Graphics g) {
        super.paintComponent(g);
        this.setBackground(Color.WHITE);

        Data.header.paintIcon(this, g, 25, 25);

        g.setColor(Color.black);
        g.fillRect(25, 75, 850, 600);

        g.setColor(Color.white);
        g.setFont(new Font("微软雅黑", Font.BOLD, 20));
        g.drawString("长度:" + length, 758, 45);
        g.drawString("积分:" + score, 758, 70);

        if (fx.equals("R")) {
            Data.right.paintIcon(this, g, snakeX[0], snakeY[0]);
        } else if (fx.equals("L")) {
            Data.left.paintIcon(this, g, snakeX[0], snakeY[0]);
        } else if (fx.equals("U")) {
            Data.up.paintIcon(this, g, snakeX[0], snakeY[0]);
        } else if (fx.equals("D")) {
            Data.down.paintIcon(this, g, snakeX[0], snakeY[0]);
        }

        for (int i = 1; i < length; i++) {
            Data.body.paintIcon(this, g, snakeX[i], snakeY[i]);
        }

        Data.food.paintIcon(this, g, foodX, foodY);

        if (!start) {
            g.setColor(Color.white);
            g.setFont(new Font("微软雅黑", Font.BOLD, 40));
            g.drawString("按下空格开始游戏", 290, 370);
        }

        if (fail) {
            g.setColor(Color.red);
            g.setFont(new Font("微软雅黑", Font.BOLD, 40));
            g.drawString("失败,按下空格重新开始", 260, 370);
        }
    }

    @Override
    public void keyPressed(KeyEvent e) {
        int keyCode = e.getKeyCode();
        if (keyCode == KeyEvent.VK_SPACE) {
            if (fail) {
                //fail = false;
                init();
            } else {
                start = !start;
            }
            repaint();
        }

        if (start && !fail) {
            if (keyCode == KeyEvent.VK_UP) {
                fx = "U";
            } else if (keyCode == KeyEvent.VK_DOWN) {
                fx = "D";
            } else if (keyCode == KeyEvent.VK_LEFT) {
                fx = "L";
            } else if (keyCode == KeyEvent.VK_RIGHT) {
                fx = "R";
            }
        }
    }


    @Override
    public void actionPerformed(ActionEvent e) {
        //timer.start();
        if (start && !fail) {
            if (snakeX[0] == foodX && snakeY[0] == foodY) {
                length++;
                score += 10;
                foodX = 25 + 25 * random.nextInt(34);
                foodY = 75 + 25 * random.nextInt(24);
            }

            for (int i = length - 1; i > 0; i--) {
                snakeX[i] = snakeX[i - 1];
                snakeY[i] = snakeY[i - 1];
            }

            if (fx.equals("R")) {
                snakeX[0] = snakeX[0] + 25;
                if (snakeX[0] > 850) {
                    snakeX[0] = 25;
                }
            } else if (fx.equals("L")) {
                snakeX[0] = snakeX[0] - 25;
                if (snakeX[0] < 25) {
                    snakeX[0] = 850;
                }
            } else if (fx.equals("U")) {
                snakeY[0] = snakeY[0] - 25;
                if (snakeY[0] < 75) {
                    snakeY[0] = 650;
                }
            } else if (fx.equals("D")) {
                snakeY[0] = snakeY[0] + 25;
                if (snakeY[0] > 650) {
                    snakeY[0] = 75;
                }
            }

            for (int i = 1; i < length; i++) {
                if (snakeX[0] == snakeX[i] && snakeY[0] == snakeY[i]) {
                    fail = true;
                }
            }
            repaint();
        }
    }

    @Override
    public void keyReleased(KeyEvent e) {

    }

    @Override
    public void keyTyped(KeyEvent e) {

    }
}
